# 验证MNIST数据集加载是否成功。
# MNIST数据集可以从网络下载，但是有时受网络限制无法得到。
# 这里将得到的数据保存在pkl文件中，需要的时候加载进入，并返回给需要的程序。

import sys,os
sys.path.append(os.pardir)
import numpy as np
from dataset.mnist import load_mnist
from PIL import Image 

# 使用PIL库验证加载的图像数据
def img_show(img):
    pil_img =Image.fromarray(np.uint8(img))
    pil_img.show()

(x_train,t_train),(x_test,t_test) = load_mnist(flatten=True,normalize=False)

img = x_train[0]
label = t_train[0]
print(label) # 5

print(img.shape) # 784
img = img.reshape(28,28)
print(img.shape) # (28,28)

img_show(img)

